#include <catch2/catch.hpp>

#include "004_medium.hpp"

using namespace jianzhi_004;

TEST_CASE("004", "[JIANZHI][004]")
{

std::vector< std::vector<int> > matrix{
    {1,   4,  7, 11, 15},
    {2,   5,  8, 12, 19},
    {3,   6,  9, 16, 22},
    {10, 13, 14, 17, 24},
    {18, 21, 23, 26, 30} };
CHECK(Solution().findNumberIn2DArray(matrix, 5)  == true);
CHECK(Solution().findNumberIn2DArray(matrix, 20) == false);

}
